NCCL_IB_DISABLE=1
device=0
run_name="test_4_21_1e-5"
CUDA_VISIBLE_DEVICES=${device}
nohup python train.py \
  --dataset mvtec \
  --train_data_path /mnt/user/wangfeng25/paper/mvtec_ad \
  --features_list 6 12 18 24 \
  --image_size 518 \
  --batch_size 8 \
  --print_freq 1 \
  --epoch 15 \
  --save_freq 1 \
  --save_path /mnt/user/wangfeng25/output \ > $run_name.log 2>&1 &

